# 获取新的令牌：https://help.aliyun.com/document_detail/611472.html?spm=a2c4g.2399481.0.0
from getpass import getpass
import os

from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_community.llms import Tongyi

# API KEY: sk-7a31e7bee6e8465ca00e6bdb375dc4cc
# 归属账号 1531421489726557
DASHSCOPE_API_KEY = r"sk-7a31e7bee6e8465ca00e6bdb375dc4cc"
os.environ["DASHSCOPE_API_KEY"] = DASHSCOPE_API_KEY

template = """
问题：{question}
回答：我们一步一步的思考。
"""
# 提示词 Prompt模板
prompt = ChatPromptTemplate.from_messages([
    ("system", "你是一个文人、诗人、历史学家、音乐家。"),
    ("user", "{input}")
])

output_parser = StrOutputParser()

llm = Tongyi()

chain = prompt | llm | output_parser

question= "请推荐一首最具中国特色和中国韵味的歌曲，并格式化展示格式歌词"
print("正在思考问题：",question)
result = chain.invoke({"input": question})
print("结果", result)
